from . import layers
from . import modeling_utils

from . import modeling_seq2seq

from . import modeling_gpt2
from . import modeling_bert
from . import modeling_albert
from . import modeling_roberta


def set_logger(logger):
    from ..utils import to_set_logger
    need_set_list = [layers,
                     modeling_utils,
                     modeling_seq2seq,
                     modeling_gpt2,
                     modeling_bert,
                     modeling_albert,
                     modeling_roberta]
    to_set_logger(logger=logger, m_list=need_set_list)


